package com.common.excel;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.support.ExcelTypeEnum;
import com.common.excel.listener.EasyExcelMapListener;
import com.common.excel.listener.EasyExcelSaveListener;
import com.common.excel.service.ExcelPoiService;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.time.DateFormatUtils;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.multipart.MultipartFile;

import java.io.*;
import java.text.DecimalFormat;
import java.util.*;


/**
 * excel文件导出导入工具
 *
 * @author zhou_xh
 * @date 2021-1-16 14:18
 */
public class ExcelImportUtils {
    private static final Logger LOGGER = LoggerFactory.getLogger(ExcelImportUtils.class);
    /**
     * 常量E
     */
    private final static String E = "E";

    /**
     * 读取excel文件，并保存数据
     *
     * @param rowNum     从哪一行开始读取数据
     * @param file       excel文件
     * @param excelSheet 读取的表单
     * @param listener   保存的监听器
     * @return
     */
    public static boolean importSaveData(File file, int excelSheet, int rowNum, EasyExcelSaveListener listener) {
        try {
            EasyExcel.read(new FileInputStream(file), listener).sheet(excelSheet).headRowNumber(rowNum).doRead();
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            LOGGER.info("excel文件读取出现问题：" + e.getLocalizedMessage());
            return false;
        }
    }

    public static boolean importSaveData(MultipartFile file, int excelSheet, int rowNum, EasyExcelSaveListener listener) {
        try {
            EasyExcel.read(file.getInputStream(), listener).sheet(excelSheet).headRowNumber(rowNum).doRead();
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            LOGGER.info("excel文件读取出现问题：" + e.getLocalizedMessage());
            return false;
        }
    }

    /**
     * 通过EasyExcel读取excel的数据
     *
     * @param file       File文件
     * @param rowNum     从哪一行开始读取数据
     * @param excelSheet 读取的表单
     * @return excel中的数据
     */

    public static List<Map<Integer, String>> importData(File file, int excelSheet, int rowNum) {
        try {
            EasyExcelMapListener listener = new EasyExcelMapListener();
            EasyExcel.read(new FileInputStream(file), listener).sheet(excelSheet).headRowNumber(rowNum).doRead();
            return listener.getData();
        } catch (Exception e) {
            e.printStackTrace();
            LOGGER.info("excel文件读取出现问题：" + e.getLocalizedMessage());
            return new ArrayList<>();
        }
    }

    /**
     * 上传文件读取excel
     * 通过EasyExcel读取excel的数据
     *
     * @param rowNum     从哪一行开始读取数据
     * @param file       上传的文件
     * @param excelSheet 读取的表单
     * @return excel中的数据
     */
    public static List<Map<Integer, String>> importData(MultipartFile file, int excelSheet, int rowNum) {
        try {
            EasyExcelMapListener listener = new EasyExcelMapListener();
            EasyExcel.read(file.getInputStream(), listener).sheet(excelSheet).headRowNumber(rowNum).doRead();
            return listener.getData();
        } catch (Exception e) {
            e.printStackTrace();
            LOGGER.info("excel文件读取出现问题：" + e.getLocalizedMessage());
            return new ArrayList<>();
        }
    }

    /**
     * 获取excel里面的数据
     *
     * @param file     文件
     * @param cell     模板的列数
     * @return 钥匙数据集合
     */
    public static <T> List<T> importData(ExcelPoiService<T> excelInterface, File file, int cell) {
        return importData(excelInterface, file, 0, cell);
    }

    /**
     * 获取excel里面的数据
     *
     * @param file       文件
     * @param excelSheet 表单
     * @param cell       模板的列数
     * @return 钥匙数据集合
     */
    public static <T> List<T> importData(ExcelPoiService<T> excelInterface, File file, int excelSheet, int cell) {
        try {
            return importData(excelInterface, file.getName(), new FileInputStream(file), excelSheet, cell);
        } catch (FileNotFoundException e) {
            LOGGER.error("excel文件读取出现问题:" + e.getLocalizedMessage());
            return new ArrayList<>();
        }
    }

    /**
     * 默认读取第0个表单
     */
    public static <T> List<T> importData(ExcelPoiService<T> excelInterface, String fileName, InputStream is, int cell) {
        return importData(excelInterface, fileName, is, 0, cell);
    }

    /**
     * 获取excel里面的数据
     *
     * @param fileName   文件名
     * @param is         文件
     * @param excelSheet 读取的excel表单
     * @param cell       模板的列数
     * @return 钥匙数据集合
     * <p>
     * 使用
     * importData((s, i, u) -> {
     * OrgKeyInfo info = new OrgKeyInfo();
     * // 添加钥匙主键
     * info.setId(IdUtil.simpleUUID());
     * // 添加钥匙名称
     * info.setKeyName(value(s, i, 0));
     * // 添加钥匙编码
     * info.setKeyCode(value(s, i, 1));
     * // 添加钥匙数量
     * String number = value(s, i, 2);
     * if (StringUtils.isNotBlank(number)) {
     * info.setKeyNumber((int) Float.parseFloat(number));
     * }
     * // 添加钥匙备注
     * info.setRemark(value(s, i, 3));
     * // 添加默认值
     * info.setBorrowNumber(0);
     * info.setBorrow(false);
     * MetaDataInfoUtil.metaDataInsert(info, u);
     * return info;
     * }, fileName, is, ORG_KEY_CELL, userInfo);
     * </p>
     */
    public static <T> List<T> importData(ExcelPoiService<T> excelInterface, String fileName, InputStream is, int excelSheet, int cell) {
        Workbook wb;
        Sheet sheet;
        List<T> list = new ArrayList<>();
        try {
            if (StringUtils.isBlank(fileName)) {
                throw new NullPointerException("导入文档为空!");
            } else {
                if (fileName.toLowerCase().endsWith(ExcelTypeEnum.XLS.getValue())) {
                    wb = new HSSFWorkbook(is);
                } else if (fileName.toLowerCase().endsWith(ExcelTypeEnum.XLSX.getValue())) {
                    wb = new XSSFWorkbook(is);
                } else {
                    return list;
                }
            }
            sheet = wb.getSheetAt(excelSheet);
            //总行数
            int num1 = sheet.getLastRowNum() + 1;
            //总列数
            int num2 = sheet.getRow(0).getPhysicalNumberOfCells();
            if (num2 != cell) {
                LOGGER.info("表单列数为：" + num2 + "，但是传入的cell=" + cell + "。与实际列数不同");
                return list;
            }
            for (int i = 1; i < num1; i++) {
                T t = excelInterface.addData(sheet, i);
                list.add(t);
            }

        } catch (Exception e) {
            e.printStackTrace();
        }
        return list;
    }

    /**
     * 获取单元格值
     *
     * @param row  行号
     * @param cell 列号
     * @return 单元格数据
     */
    public static String value(Sheet sheet, int row, int cell) {
        return getCellFormatValue(sheet.getRow(row).getCell(cell));
    }

    /**
     * 根据Cell类型设置数据
     *
     * @param cell 单元格
     * @return 字符串，单元格内的数据
     */
    private static String getCellFormatValue(Cell cell) {
        String cellValue = "";
        if (cell != null) {
            // 判断当前Cell的Type   cell.getCellType()
            switch (cell.getCellType()) {
                // 如果当前Cell的Type为NUMERIC
                case _NONE:
                    break;
                case NUMERIC: {
                    if (!String.valueOf(cell.getNumericCellValue()).contains(E)) {
                        cellValue = String.valueOf(cell.getNumericCellValue());
                    } else {
                        cellValue = new DecimalFormat("#").format(cell.getNumericCellValue());
                    }
                    //如果为Number类型的Date，转换为Date
                    if (DateUtil.isCellDateFormatted(cell)) {
                        Date date = cell.getDateCellValue();
                        cellValue = DateFormatUtils.format(date, "yyyy-MM-dd HH:mm:ss");
                    }
                    break;
                }
                case FORMULA: {
                    // 判断当前的cell是否为Date
                    if (DateUtil.isCellDateFormatted(cell)) {
                        Date date = cell.getDateCellValue();
                        cellValue = DateFormatUtils.format(date, "yyyy-MM-dd HH:mm:ss");
                    } else {
                        // 如果是纯数字
                        // 取得当前Cell的数值
                        cellValue = String.valueOf(cell.getNumericCellValue());
                    }
                    break;
                }
                // 如果当前Cell的Type为STRING
                case STRING:
                    // 取得当前的Cell字符串
                    cellValue = cell.getRichStringCellValue().getString();
                    break;
                default:// 默认的Cell值
                    cellValue = "";
            }
        } else {
            cellValue = "";
        }
        return cellValue;
    }
}
